#!/usr/bin/env python
# -*- coding: utf-8 -*

"""
@author: coldwind
"""

with open('data/train_v1.dat', 'r') as input_file, \
        open('data/train_v1_shuffle.dat', 'w') as train_shuffle, \
        open('data/test_v1_shuffle.dat', 'w') as test_shuffle:
    count = 0
    for line in input_file:
        count += 1
        if count % 200 == 0:
            test_shuffle.write(line)
        else:
            train_shuffle.write(line)
